# 数据库配置信息
DB_CONFIG = {
    "host": "127.0.0.1",        # 数据库主机地址
    "user": "postgres",         # 数据库用户名
    "password": "postgres",     # 数据库密码
    "database": "stockdb",      # 数据库名称
    "port": 5432                # 数据库端口号
}

# 特征列顺序，用于模型训练时保持特征的一致性
FEATURE_ORDER = [
    "open", "high", "low", "close", "volume",      # 基础行情数据
    "ret_1d", "ret_5d",                            # 收益率特征
    "ma5", "ma10",                                 # 移动平均线
    "rsi14",                                       # 相对强弱指数
    "vol_ma5",                                     # 成交量移动平均
    "sse_ret", "csi300_ret", "industry_ret"        # 市场和行业基准收益
]

# 模型存储目录
MODEL_DIR = "model"